##############################################################
# Copyright (c) 2018-present, Facebook, Inc.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
##############################################################

# My awesome color map
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import numpy as np


def colormap(rgb=False):
    color_list = np.array([0.000, 0.447, 0.741,
                            0.850, 0.325, 0.098,
                            0.929, 0.694, 0.125,
                            0.494, 0.184, 0.556,
                            0.466, 0.674, 0.188,
                            0.301, 0.745, 0.933,
                            0.635, 0.078, 0.184,
                            0.300, 0.300, 0.300,
                            0.600, 0.600, 0.600,
                            1.000, 0.000, 0.000,
                            1.000, 0.500, 0.000,
                            0.749, 0.749, 0.000,
                            0.000, 1.000, 0.000,
                            0.000, 0.000, 1.000,
                            0.667, 0.000, 1.000,
                            0.333, 0.333, 0.000,
                            0.333, 0.667, 0.000,
                            0.333, 1.000, 0.000,
                            0.667, 0.333, 0.000,
                            0.667, 0.667, 0.000,
                            0.667, 1.000, 0.000,
                            1.000, 0.333, 0.000,
                            1.000, 0.667, 0.000,
                            1.000, 1.000, 0.000,
                            0.000, 0.333, 0.500,
                            0.000, 0.667, 0.500,
                            0.000, 1.000, 0.500,
                            0.333, 0.000, 0.500,
                            0.333, 0.333, 0.500,
                            0.333, 0.667, 0.500,
                            0.333, 1.000, 0.500,
                            0.667, 0.000, 0.500,
                            0.667, 0.333, 0.500,
                            0.667, 0.667, 0.500,
                            0.667, 1.000, 0.500,
                            1.000, 0.000, 0.500,
                            1.000, 0.333, 0.500,
                            1.000, 0.667, 0.500,
                            1.000, 1.000, 0.500,
                            0.000, 0.333, 1.000,
                            0.000, 0.667, 1.000,
                            0.000, 1.000, 1.000,
                            0.333, 0.000, 1.000,
                            0.333, 0.333, 1.000,
                            0.333, 0.667, 1.000,
                            0.333, 1.000, 1.000,
                            0.667, 0.000, 1.000,
                            0.667, 0.333, 1.000,
                            0.667, 0.667, 1.000,
                            0.667, 1.000, 1.000,
                            1.000, 0.000, 1.000,
                            1.000, 0.333, 1.000,
                            1.000, 0.667, 1.000,
                            0.167, 0.000, 0.000,
                            0.333, 0.000, 0.000,
                            0.500, 0.000, 0.000,
                            0.667, 0.000, 0.000,
                            0.833, 0.000, 0.000,
                            1.000, 0.000, 0.000,
                            0.000, 0.167, 0.000,
                            0.000, 0.333, 0.000,
                            0.000, 0.500, 0.000,
                            0.000, 0.667, 0.000,
                            0.000, 0.833, 0.000,
                            0.000, 1.000, 0.000,
                            0.000, 0.000, 0.167,
                            0.000, 0.000, 0.333,
                            0.000, 0.000, 0.500,
                            0.000, 0.000, 0.667,
                            0.000, 0.000, 0.833,
                            0.000, 0.000, 1.000,
                            0.000, 0.000, 0.000,
                            0.143, 0.143, 0.143,
                            0.286, 0.286, 0.286,
                            0.429, 0.429, 0.429,
                            0.571, 0.571, 0.571,
                            0.714, 0.714, 0.714,
                            0.857, 0.857, 0.857,
                            1.000, 1.000, 1.000]).astype(np.float32)
    color_list = color_list.reshape((-1, 3)) * 255
    if not rgb:
        color_list = color_list[:, ::-1]
    return color_list
